# -*- coding: utf-8 -*-
"""
@Time    : 2024/6/26 10:56 
@Author  : ZhangShenao 
@File    : 1.PromptTemplate基础用法.py
@Desc    : PromptTemplate基础用法
"""
from langchain_core.prompts import PromptTemplate

# 创建PromptTemplate
prompt_template = PromptTemplate.from_template('What is the capital of {country}?')
print(f'prompt_template: {prompt_template}')

# 调用创建PromptTemplate.invoke()函数,传递变量值,生成Prompt
prompt_value = prompt_template.invoke({
    'country': 'China',
})
print(f'prompt_value: {prompt_value}')

# 调用创建PromptTemplate.format()函数,传递变量值,生成文本消息
prompt_text = prompt_template.format(country='China')
print(f'prompt_text: {prompt_text}')

# 调用创建PromptValue.to_string()函数,将Prompt转化成文本消息
prompt_value_str = prompt_value.to_string()
print(f'prompt_value_str: {prompt_value_str}')
